tensor([[[[[0.3700, 0.0000],
           [0.0000, 0.0000]],

          [[0.0000, 0.2200],
           [0.4900, 0.0000]]],


         [[[0.2300, 0.0000],
           [0.5000, 0.4900]],

          [[0.2100, 0.2100],
           [0.2100, 0.2100]]]],



        [[[[0.2900, 0.0000],
           [1.0000, 0.0000]],

          [[0.5200, 0.3000],
           [0.5100, 0.0000]]],


         [[[0.3000, 0.0000],
           [0.0000, 0.5100]],

          [[0.3200, 0.3200],
           [0.3100, 0.3100]]]],



        [[[[0.1800, 0.5000],
           [0.0000, 0.5000]],

          [[0.0000, 0.2300],
           [0.0000, 0.0000]]],


         [[[0.2300, 0.5000],
           [0.5000, 0.0000]],

          [[0.2300, 0.2200],
           [0.2300, 0.2300]]]],



        [[[[0.1600, 0.5000],
           [0.0000, 0.5000]],

          [[0.4800, 0.2500],
           [0.0000, 1.0000]]],


         [[[0.2400, 0.5000],
           [0.0000, 0.0000]],

          [[0.2500, 0.2400],
           [0.2500, 0.2500]]]]], grad_fn=<RoundBackward1>)